/*
 * Copyright (C) 2016-2023 ActionTech.
 * based on code by MyCATCopyrightHolder Copyright (c) 2013, OpenCloudDB/MyCAT.
 * License: http://www.gnu.org/licenses/gpl.html GPL version 2 or higher.
 */
package com.actiontech.dble.util;

import java.nio.charset.Charset;
import java.util.Date;

public final class ByteUtil {
    private ByteUtil() {
    }

    /**
     * compare to number or dicamal ascii byte array, for number :123456 ,store
     * to array [1,2,3,4,5,6]
     *
     * @param b1
     * @param b2
     * @return -1 means b1 < b2, or 0 means b1=b2 else return 1
     */
    public static int compareNumberByte(byte[] b1, byte[] b2) {
        if (b1 == null || b1.length == 0) {
            return -1;
        } else if (b2 == null || b2.length == 0) {
            return 1;
        }
        boolean isNegetive = b1[0] == 45 || b2[0] == 45;
        if (!isNegetive && b1.length != b2.length) {
            return b1.length - b2.length;
        }
        int len = Math.min(b1.length, b2.length);
        int result = 0;
        int index = -1;
        for (int i = 0; i < len; i++) {
            int b1val = b1[i];
            int b2val = b2[i];
            if (b1val > b2val) {
                result = 1;
                index = i;
                break;
            } else if (b1val < b2val) {
                index = i;
                result = -1;
                break;
            }
        }
        if (index == 0) {
            // first byte compare
            return result;
        } else {
            if (b1.length != b2.length) {

                int lenDelta = b1.length - b2.length;
                return isNegetive ? 0 - lenDelta : lenDelta;

            } else {
                return isNegetive ? 0 - result : result;
            }
        }
    }

    public static byte[] compareNumberArray2(byte[] b1, byte[] b2, int order) {
        if (b1.length <= 0 && b2.length > 0) {
            return b2;
        }
        if (b1.length > 0 && b2.length <= 0) {
            return b1;
        }
        int len = b1.length > b2.length ? b1.length : b2.length;
        for (int i = 0; i < len; i++) {
            if (b1[i] != b2[i]) {
                if (order == 1) {
                    return ((b1[i] & 0xff) - (b2[i] & 0xff)) > 0 ? b1 : b2;
                } else {
                    return ((b1[i] & 0xff) - (b2[i] & 0xff)) > 0 ? b2 : b1;
                }
            }
        }

        return b1;
    }

    public static byte[] getBytes(short data) {
        byte[] bytes = new byte[2];
        bytes[0] = (byte) (data & 0xff);
        bytes[1] = (byte) ((data & 0xff00) >> 8);
        return bytes;
    }

    public static byte[] getBytes(char data) {
        byte[] bytes = new byte[2];
        bytes[0] = (byte) (data);
        bytes[1] = (byte) (data >> 8);
        return bytes;
    }

    public static byte[] getBytes(int data) {
        byte[] bytes = new byte[4];
        bytes[0] = (byte) (data & 0xff);
        bytes[1] = (byte) ((data & 0xff00) >> 8);
        bytes[2] = (byte) ((data & 0xff0000) >> 16);
        bytes[3] = (byte) ((data & 0xff000000) >> 24);
        return bytes;
    }

    public static byte[] getBytes(long data) {
        byte[] bytes = new byte[8];
        bytes[0] = (byte) (data & 0xff);
        bytes[1] = (byte) ((data >> 8) & 0xff);
        bytes[2] = (byte) ((data >> 16) & 0xff);
        bytes[3] = (byte) ((data >> 24) & 0xff);
        bytes[4] = (byte) ((data >> 32) & 0xff);
        bytes[5] = (byte) ((data >> 40) & 0xff);
        bytes[6] = (byte) ((data >> 48) & 0xff);
        bytes[7] = (byte) ((data >> 56) & 0xff);
        return bytes;
    }

    public static byte[] getBytes(float data) {
        int intBits = Float.floatToIntBits(data);
        return getBytes(intBits);
    }

    public static byte[] getBytes(double data) {
        long intBits = Double.doubleToLongBits(data);
        return getBytes(intBits);
    }

    public static byte[] getBytes(String data, String charsetName) {
        Charset charset = Charset.forName(charsetName);
        return data.getBytes(charset);
    }

    public static byte[] getBytes(String data) {
        return getBytes(data, "GBK");
    }

    public static byte[] getBytes(Date date, boolean isTime) {
        if (isTime) {
            return getBytesFromTime(date);
        } else {
            return getBytesFromDate(date);
        }
    }

    public static short getShort(byte[] bytes) {
        return Short.parseShort(new String(bytes));
    }

    public static char getChar(byte[] bytes) {
        return (char) ((0xff & bytes[0]) | (0xff00 & (bytes[1] << 8)));
    }

    public static int getInt(byte[] bytes) {
        return Integer.parseInt(new String(bytes));
        // return (0xff & bytes[0]) | (0xff00 & (bytes[1] << 8)) | (0xff0000 &
        // (bytes[2] << 16)) | (0xff000000 & (bytes[3] << 24));
    }

    public static long getLong(byte[] bytes) {
        return Long.parseLong(new String(bytes));
        // return(0xffL & (long)bytes[0]) | (0xff00L & ((long)bytes[1] << 8)) |
        // (0xff0000L & ((long)bytes[2] << 16)) | (0xff000000L & ((long)bytes[3]
        // << 24))
        // | (0xff00000000L & ((long)bytes[4] << 32)) | (0xff0000000000L &
        // ((long)bytes[5] << 40)) | (0xff000000000000L & ((long)bytes[6] <<
        // 48)) | (0xff00000000000000L & ((long)bytes[7] << 56));
    }

    public static double getDouble(byte[] bytes) {
        return Double.parseDouble(new String(bytes));
    }

    public static float getFloat(byte[] bytes) {
        return Float.parseFloat(new String(bytes));
    }

    public static String getString(byte[] bytes, String charsetName) {
        return new String(bytes, Charset.forName(charsetName));
    }

    public static String getString(byte[] bytes) {
        return getString(bytes, "UTF-8");
    }

    public static String getDate(byte[] bytes) {
        return new String(bytes);
    }

    public static String getTime(byte[] bytes) {
        return new String(bytes);
    }

    public static String getTimestmap(byte[] bytes) {
        return new String(bytes);
    }

    private static byte[] getBytesFromTime(Date date) {
        int day = 0;
        int hour = DateUtil.getHour(date);
        int minute = DateUtil.getMinute(date);
        int second = DateUtil.getSecond(date);
        int microSecond = DateUtil.getMicroSecond(date);
        byte[] bytes = null;
        byte[] tmp = null;
        if (hour == 0 && minute == 0 && second == 0 && microSecond == 0) {
            bytes = new byte[1];
            bytes[0] = (byte) 0;
        } else if (microSecond == 0) {
            bytes = new byte[1 + 8];
            bytes[0] = (byte) 8;
            bytes[1] = (byte) 0; // is_negative (1) -- (1 if minus, 0 for plus)
            tmp = getBytes(day);
            bytes[2] = tmp[0];
            bytes[3] = tmp[1];
            bytes[4] = tmp[2];
            bytes[5] = tmp[3];
            bytes[6] = (byte) hour;
            bytes[7] = (byte) minute;
            bytes[8] = (byte) second;
        } else {
            bytes = new byte[1 + 12];
            bytes[0] = (byte) 12;
            bytes[1] = (byte) 0; // is_negative (1) -- (1 if minus, 0 for plus)
            tmp = getBytes(day);
            bytes[2] = tmp[0];
            bytes[3] = tmp[1];
            bytes[4] = tmp[2];
            bytes[5] = tmp[3];
            bytes[6] = (byte) hour;
            bytes[7] = (byte) minute;
            bytes[8] = (byte) second;
            tmp = getBytes(microSecond);
            bytes[9] = tmp[0];
            bytes[10] = tmp[1];
            bytes[11] = tmp[2];
            bytes[12] = tmp[3];
        }
        return bytes;
    }

    private static byte[] getBytesFromDate(Date date) {
        int year = DateUtil.getYear(date);
        int month = DateUtil.getMonth(date);
        int day = DateUtil.getDay(date);
        int hour = DateUtil.getHour(date);
        int minute = DateUtil.getMinute(date);
        int second = DateUtil.getSecond(date);
        int microSecond = DateUtil.getMicroSecond(date);
        byte[] bytes = null;
        byte[] tmp = null;
        if (year == 0 && month == 0 && day == 0 && hour == 0 && minute == 0 && second == 0 && microSecond == 0) {
            bytes = new byte[1];
            bytes[0] = (byte) 0;
        } else if (hour == 0 && minute == 0 && second == 0 && microSecond == 0) {
            bytes = new byte[1 + 4];
            bytes[0] = (byte) 4;
            tmp = getBytes((short) year);
            bytes[1] = tmp[0];
            bytes[2] = tmp[1];
            bytes[3] = (byte) month;
            bytes[4] = (byte) day;
        } else if (microSecond == 0) {
            bytes = new byte[1 + 7];
            bytes[0] = (byte) 7;
            tmp = getBytes((short) year);
            bytes[1] = tmp[0];
            bytes[2] = tmp[1];
            bytes[3] = (byte) month;
            bytes[4] = (byte) day;
            bytes[5] = (byte) hour;
            bytes[6] = (byte) minute;
            bytes[7] = (byte) second;
        } else {
            bytes = new byte[1 + 11];
            bytes[0] = (byte) 11;
            tmp = getBytes((short) year);
            bytes[1] = tmp[0];
            bytes[2] = tmp[1];
            bytes[3] = (byte) month;
            bytes[4] = (byte) day;
            bytes[5] = (byte) hour;
            bytes[6] = (byte) minute;
            bytes[7] = (byte) second;
            tmp = getBytes(microSecond);
            bytes[8] = tmp[0];
            bytes[9] = tmp[1];
            bytes[10] = tmp[2];
            bytes[11] = tmp[3];
        }
        return bytes;
    }

}
